package com.gzz.orchid.common.boot.configuraction;

import jakarta.annotation.PostConstruct;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.ReturnedMessage;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Configuration;

/**
 * @author higmd
 */
@Slf4j
@Configuration
public class RabbitConfirmConfig implements RabbitTemplate.ConfirmCallback, RabbitTemplate.ReturnsCallback {
    @Autowired
    private RabbitTemplate rabbitTemplate;

    @PostConstruct
    public void init() {
        //设置开启Mandatory,才能触发回调函数,无论消息推送结果怎么样都强制调用回调函数
        rabbitTemplate.setMandatory(true);
        rabbitTemplate.setConfirmCallback(this);
        rabbitTemplate.setReturnsCallback(this);
    }

    /**
     * 实现confirm回调,
     * 发送和没发送到exchange,都触发 (前提: 确认模式开启：yml中publisher-confirm-type: correlated)
     * CorrelationData数据可以在rabbitTemplate.convertAndSend时传入 并这种CorrelationData的setId参数,回调时能取到
     */
    @Override
    public void confirm(CorrelationData correlationData, boolean ack, String cause) {
        // 参数说明:
        // correlationData: 相关数据,可以在发送消息时,进行设置该参数
        // ack: 结果
        // cause: 原因
        if (ack) {
            log.info("消息发送到交换机成功: correlationData={}", correlationData);
            log.info("【ConfirmCallback】消息已经送达Exchange，ack已发");

        } else {
            log.error("消息发送到交换机失败: correlationData={}, cause={}", correlationData, cause);
            // 处理失败逻辑，如重试、告警等
            log.warn("【ConfirmCallback】消息没有送达Exchange");
            // todo 做一些处理,让消息再次发送。 消息缓存或入库，邮件提醒运维

        }
    }

    /**
     * 实现return回调:当消息发送给Exchange后,Exchange路由到Queue失败时 才会执行 ReturnCallBack
     * (前提:退回模式开启：yml中publisher-returns: true)
     */
    @Override
    public void returnedMessage(ReturnedMessage returned) {
        log.error("消息路由到队列失败: exchange={}, routingKey={}, replyCode={}, replyText={}",
                returned.getExchange(),
                returned.getRoutingKey(),
                returned.getReplyCode(),
                returned.getReplyText());
        log.error("消息内容: message={}", new String(returned.getMessage().getBody()));
        // 处理失败逻辑，如重试、告警等

    }
}
